package lc.q201_250;

import java.util.HashSet;
import java.util.Set;

/**
 * 编写一个算法来判断一个数 n 是不是快乐数。
 *
 * 「快乐数」定义为：对于一个正整数，每一次将该数替换为它每个位置上的数字的平方和，然后重复这个过程直到这个数变为 1，也可能是 无限循环 但始终变不到 1。如果 可以变为  1，那么这个数就是快乐数。
 *
 * 如果 n 是快乐数就返回 True ；不是，则返回 False 。
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode-cn.com/problems/happy-number
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 *
 * @author lgb
 * @date 2020/08/23
 **/
public class Q202 {
    public boolean isHappy(int n) {
        // 如果出现过该数，则不会是快乐数
        Set<Integer> store = new HashSet<>();
        while (!store.contains(n)) {
            if (n == 1) {
                return true;
            }
            store.add(n);
            n = getNext(n);
        }
        return false;
    }

    public boolean isHappy2(int n) {// 快慢指针法
        int slow = n;
        int fast = getNext(n);
        while (fast != 1 && slow != fast) {
            slow = getNext(slow);
            fast = getNext(getNext(fast));
        }
        return fast == 1;
    }

    private int getNext(int m) {// 时间复杂度为O(LOGn)
        int result = 0;
        while (m != 0) {
            result += (m % 10) * (m % 10);
            m = m / 10;
        }
        return result;
    }

    public static void main(String[] args) {
        int k = 19;
        System.out.println(new Q202().isHappy2(k));
    }
}
